cd ../..

gpus=$1
psd_lo_thd=0.10
psd_hi_thd=0.90
weight=0
# for dataset in "Y800M_random_140h" #  
# do
python -m trainer.ATSTFTTEST fit \
    --config configs/model/ATSTSED_contrastive.yaml \
    --config configs/dataset/desed_with_id.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --config configs/experiment/plugin_ATSTFrame_finetune.yaml \
    --model.exp_name="5_cl_usage/ATSTSED/stage2/prev_ckpt/400+no_warp" \
    --data.init_args.batch_sizes=[24,24,48,0,48] \
    --model.cl_decay=True \
    --model.cl_weight=${weight} \
    --trainer.devices=${gpus} \
    --trainer.strategy="ddp_find_unused_parameters_true" \
    --model.psd_label_filter=False \
    --model.psd_lo_thd=${psd_lo_thd} \
    --model.psd_hi_thd=${psd_hi_thd} \
    --model.ckpt_path="/home/shaonian/sed/pretrained_ckpts/prev_stage1.ckpt" \
    --model.sed_model.init_args.atst_train=True \
    --model.unsupervised_loss.init_args.weight=70 \
    --trainer.max_epochs=400 
# done